#!/bin/bash

# Update the package list and install necessary packages
dnf update -y

# タイムゾーンの設定
timedatectl set-timezone Asia/Tokyo

# rsyslogのインストール（システムログがjournalで保存され文字化けするため）
dnf install -y rsyslog
systemctl enable rsyslog
systemctl start rsyslog

# CloudWatch Agentのインストール
dnf install -y amazon-cloudwatch-agent
# Create the CloudWatch Agent configuration file for logging
bash -c 'cat <<EOL > /opt/aws/amazon-cloudwatch-agent/bin/config.json
{
  "logs": {
    "logs_collected": {
      "files": {
        "collect_list": [
          {
            "file_path": "/var/log/messages",
            "log_group_name": "/api-server",
            "log_stream_name": "{instance_id}",
            "retention_in_days": 7
          },
          {
            "file_path": "/var/log/api-app/*.log",
            "log_group_name": "/api-server-app",
            "log_stream_name": "{instance_id}",
            "retention_in_days": 7
          }
        ]
      }
    }
  }
}
EOL'

# Start the CloudWatch Agent
/opt/aws/amazon-cloudwatch-agent/bin/amazon-cloudwatch-agent-ctl -a fetch-config -m ec2 -c file:/opt/aws/amazon-cloudwatch-agent/bin/config.json -s

# デフォルトでenableだが念のため明示的にenableしておく
systemctl enable amazon-cloudwatch-agent

# ログファイル用のディレクトリ作成
mkdir /var/log/api-app
chown -R ec2-user:ec2-user /var/log/api-app

# nodejsのインストールの準備
dnf install -y gcc-c++ make

# マルチパート形式
Content-Type: multipart/mixed; boundary="//"
MIME-Version: 1.0

--//
Content-Type: text/cloud-config; charset="us-ascii"
MIME-Version: 1.0
Content-Transfer-Encoding: 7bit
Content-Disposition: attachment; filename="cloud-config.txt"

#cloud-config
cloud-init directives

--//
Content-Type: text/x-shellscript; charset="us-ascii"
MIME-Version: 1.0
Content-Transfer-Encoding: 7bit
Content-Disposition: attachment; filename="userdata.txt"

#!/bin/bash
if ! su - ec2-user -c "command -v node" > /dev/null 2>&1; then
  su - ec2-user -c "curl -sL https://rpm.nodesource.com/setup_20.x | sudo -E bash -"
fi
--//--

# nodejsのインストール
dnf install -y nodejs

# ディレクトリ作成
mkdir /home/ec2-user/tidb-connection-api
chown -R ec2-user:ec2-user /home/ec2-user/tidb-connection-api

import express from 'express';
import mysql from 'mysql2/promise';
import dotenv from 'dotenv';
import fs from 'fs';
import pino from 'pino';

// ログファイルを指定
const logFile = fs.createWriteStream('/var/log/api-app/app.log', { flags: 'a' });
const logger = pino(logFile);

const app = express();

dotenv.config();

// データベース接続設定
const dbConfig = {
    host: process.env.TIDB_HOST,
    port: process.env.TIDB_PORT,
    user: process.env.TIDB_USER,
    password: process.env.TIDB_PASSWORD,
    database: process.env.TIDB_DATABASE,
    ssl: process.env.TIDB_ENABLE_SSL === 'true' ? {
        // SSLの設定
        minVersion: 'TLSv1.2',
        ca: fs.readFileSync(process.env.TIDB_CA_PATH)
    } : null,
};
// Gatlingから受けるポート
const PORT = 3000;

app.get('/read', async (req, res) => {
    let connection;
    try {
        // データベースに接続
        connection = await mysql.createConnection(dbConfig);
        logger.info('Connection has been established successfully.');
        // クエリを実行
        const [results, fields] = await connection.execute('SHOW DATABASES');
        // 結果を返す
        res.json({
            results: results,
            fields: fields
        });
        logger.info('Read Query OK');
    } catch (err) {
        logger.error('Error executing query:', err.message);
        res.status(500).json({ error: 'Database query failed' });
    } finally {
        if (connection) {
            // 接続を閉じる
            await connection.end();
        }
    }
});

app.use((req, res, next) => {
    logger.error('Not Found:', err.message);
    res.status(404).json({ message: 'Not Found' });
});

app.listen(PORT, () => {
    logger.info(`Server is running on port ${PORT}`);
});

TIDB_HOST=tidb.xxxxxxxxx.clusters.tidb-cloud.com
TIDB_PORT=4000
TIDB_USER=root
TIDB_PASSWORD=xxxxxxxx
TIDB_DATABASE=test
TIDB_ENABLE_SSL=true
TIDB_CA_PATH=ca.cer

#!/bin/bash

# Update the package list and install necessary packages
dnf update -y

# タイムゾーンの設定
timedatectl set-timezone Asia/Tokyo

# JDKのインストール
dnf install -y java-21-amazon-corretto-devel

# Download and extract Gatling
GATLING_VERSION="3.11.5"
wget https://repo1.maven.org/maven2/io/gatling/highcharts/gatling-charts-highcharts-bundle/${GATLING_VERSION}/gatling-charts-highcharts-bundle-${GATLING_VERSION}.zip
unzip gatling-charts-highcharts-bundle-${GATLING_VERSION}.zip
cp -r gatling-charts-highcharts-bundle-${GATLING_VERSION} /home/ec2-user/gatling
chown -R ec2-user:ec2-user /home/ec2-user/gatling

package simulation;

import io.gatling.javaapi.core.CoreDsl.*;
import io.gatling.javaapi.http.HttpDsl.*;
import io.gatling.javaapi.core.*;
import io.gatling.javaapi.http.*;

import java.time.Duration;

import static io.gatling.javaapi.core.CoreDsl.*;
import static io.gatling.javaapi.http.HttpDsl.*;

public class ApiSimulation extends Simulation {

    HttpProtocolBuilder httpProtocol = http
        .baseUrl("http://{api-server-private-subnet}:3000")
        .acceptHeader("application/json");

    ScenarioBuilder scn = scenario("TiDB Request Scenario")
        .exec(
            http("Request to API")
                .get("/read")
                .check(status().is(200))
        );

    {
        setUp(
            scn.injectOpen(atOnceUsers(10))
        ).protocols(httpProtocol);
    }
}

node index.mjs

cd gatling
./mvnw gatling:test -Dgatling.simulationClass=simulations.ApiSimulation